import os


def rename_files_in_dir(directory, old_extension, new_extension):
    """
    将指定目录下所有指定扩展名的文件更改为新的扩展名。

    参数：
        directory (str): 源目录路径。
        old_extension (str): 当前文件扩展名（包含“.”）。
        new_extension (str): 新的文件扩展名（包含“.”）。
    """
    for filename in os.listdir(directory):
        # 检查是否为指定扩展名的文件且是当前目录下的文件（非子目录）
        if filename.endswith(old_extension) and os.path.isfile(os.path.join(directory, filename)):
            # 获取不带扩展名的文件名部分
            base_name = os.path.splitext(filename)[0]

            # 构造新文件名
            new_filename = base_name + new_extension

            # 获取完整路径
            old_file_path = os.path.join(directory, filename)
            new_file_path = os.path.join(directory, new_filename)

            # 更改文件扩展名（重命名文件）
            os.rename(old_file_path, new_file_path)


if __name__ == '__main__':
    rename_files_in_dir('./output', '.txt', '.rpy')
